
from decoder_mc import v_mainchain, show_mutiply_structures,get_sidechain_angles,get_1d_mc_sd


def get_seq(g_data):
    seq_lst = []
    for i in range(len(g_data)):

        data = g_data[i].detach().numpy()

        seq, mc, sd = get_1d_mc_sd(data)
        seq_lst.append(seq)
        print(seq)
    return seq_lst

import torch
import pickle    


import sys
pre  = sys.argv[1]

def get_seq_lst(kkk):
    with open(pre,"rb") as f:
        d1= pickle.load(f)
    a = torch.zeros([d1.shape[0],14,42])
    for i in range(len(d1)):
        a[i,:14,:42] = d1[i,0,:14,:42]
    g_data = a
    r_lst = get_seq(g_data)
    return r_lst

r_lst = get_seq_lst(0)
#for  r in r_lst:
#    print(r)

